Pytorch中torch.full(),torch.ones()和torch.zeros()函数解析 |
您所在的位置:网站首页 › pytorch new_ones › Pytorch中torch.full(),torch.ones()和torch.zeros()函数解析 |
一. torch.full()函数解析
1. 官网链接
torch.full(),如下图所示: 2. torch.full()函数解析 torch.full(size, fill_value, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor返回创建size大小的维度,里面元素全部填充为fill_value 3.代码举例输出形状为(2,3)的二维张量,里面元素全部填充为5 x = torch.full(size=(2,3),fill_value=5) x 输出结果如下: tensor([[5, 5, 5], [5, 5, 5]]) 二. torch.ones()函数解析 1. 官网链接torch.ones(),如下图所示: 2. torch.ones()函数解析 torch.ones(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor返回创建size大小的维度,里面元素全部填充为1 3. 代码举例 a = torch.ones(2, 3) b = torch.ones(5) c = torch.ones(size=(3,4)) a,b,c 输出结果如下: (tensor([[1., 1., 1.], [1., 1., 1.]]), tensor([1., 1., 1., 1., 1.]), tensor([[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]])) 三. torch.zeros()函数解析 1.官网链接torch.zeros(),如下图所示: 2. torch.zeros()函数解析 torch.zeros(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor返回创建size大小的维度,里面元素全部填充为0 3. 代码举例 a = torch.zeros(2, 3) b = torch.zeros(5) c = torch.zeros(size=(3,4)) a,b,c 输出结果如下: (tensor([[0., 0., 0.], [0., 0., 0.]]), tensor([0., 0., 0., 0., 0.]), tensor([[0., 0., 0., 0.], [0., 0., 0., 0.], [0., 0., 0., 0.]])) |
今日新闻 |
点击排行 |
|
推荐新闻 |
图片新闻 |
|
专题文章 |
CopyRight 2018-2019 实验室设备网 版权所有 win10的实时保护怎么永久关闭 |